package com.example.demo.config;

/**
 * @author lyc
 * @date 2019/7/27.
 */
import java.io.IOException;
import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.http.HttpMethod;
import org.springframework.stereotype.Component;

@Component
public class CorsFilter implements Filter {
    @Override
    public void doFilter(ServletRequest req, ServletResponse res, FilterChain chain) throws IOException, ServletException {
        HttpServletRequest request = (HttpServletRequest) req;
        HttpServletResponse response = (HttpServletResponse) res;
        String curOrigin = request.getHeader("Origin");
        System.out.println("###跨域过滤器->当前访问来源->"+curOrigin+"###");
        if(curOrigin == null || curOrigin.indexOf("127.0.0.1:8080")>-1){
            response.setHeader("Access-Control-Allow-Origin", "*");
        }else {
            if(curOrigin !=null)
            response.setHeader("Access-Control-Allow-Origin", curOrigin);
        }
        //放行
        if (request.getMethod().equals(HttpMethod.OPTIONS)) {
            return;
        }
        // 自定义头
        String headers = request.getHeader("Access-Control-Request-Headers");
        if (headers != null) {
            response.addHeader("Access-Control-Allow-Headers", headers);
        }

       // response.setHeader("Access-Control-Allow-Origin", "*");
        response.setHeader("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE, PUT, GET");
        response.setHeader("Access-Control-Max-Age", "3600");
        response.setHeader("Access-Control-Allow-Headers", "x-requested-with");
        chain.doFilter(req, res);
    }
    @Override
    public void init(FilterConfig filterConfig) {}

    @Override
    public void destroy() {}
}
